"""
折线图开发
"""
import json
from pyecharts.charts import Line
from pyecharts.options import TitleOpts, LabelOpts

# 打开要处理的数据
f_us = open(r"C:\Users\Lenovo\OneDrive\桌面\三种图的案例\美国.txt", "r", encoding="UTF-8")
us_data = f_us.read()    # 美国的全部内容

f_jp = open(r"C:\Users\Lenovo\OneDrive\桌面\三种图的案例\日本.txt", "r", encoding="UTF-8")
jp_data = f_jp.read()    # 日本的全部内容

f_in = open(r"C:\Users\Lenovo\OneDrive\桌面\三种图的案例\印度.txt", "r", encoding="UTF-8")
in_data = f_in.read()    # 印度的全部内容

# 去除不合JSON规范的部分
us_data = us_data.replace("jsonp_1629344292311_69436(", "")
us_data = us_data[:-2]

jp_data = jp_data.replace("jsonp_1629350871167_29498(", "")
jp_data = jp_data[:-2]

in_data = in_data.replace("jsonp_1629350745930_63180(", "")
in_data = in_data[:-2]

# JSON转Python字典
us_dict = json.loads(us_data)
print(type(us_dict))    # 检查是否转化成功
print(us_dict)

jp_dict = json.loads(jp_data)
print(type(jp_dict))    # 检查是否转化成功
print(jp_dict)

in_dict = json.loads(in_data)
print(type(in_dict))    # 检查是否转化成功
print(in_dict)


# 获取trend key
us_trend_data = us_dict["data"][0]["trend"]
jp_trend_data = jp_dict["data"][0]["trend"]
in_trend_data = in_dict["data"][0]["trend"]

# 获取日期，用于x轴，取2020年（到下标314结束）
x_data = us_trend_data["updateDate"][:314]
# 获取确诊人数，用于y轴，取2020年（到下标314结束）
us_y_data = us_trend_data["list"][0]["data"][:314]
jp_y_data = jp_trend_data["list"][0]["data"][:314]
in_y_data = in_trend_data["list"][0]["data"][:314]


# 生成图像
# 在顶部导包

# 获取折线图对象
line = Line()

# 添加x轴数据（日期），三个国家的日期都相同，导入一个即可
line.add_xaxis(x_data)
# 添加y轴确诊人数数据
line.add_yaxis("美国确诊人数", us_y_data, label_opts=LabelOpts(is_show=False))
line.add_yaxis("日本确诊人数", jp_y_data, label_opts=LabelOpts(is_show=False))
line.add_yaxis("印度确诊人数", in_y_data, label_opts=LabelOpts(is_show=False))

# 调用render方法，生成图表
line.render()

# 全局设置
line.set_global_opts(
    # 标题设置
    title_opts=TitleOpts(
        title="2020年美日印三国确诊人数对比折线图",
        pos_left="center",  # 修正为标准参数值 "center"（而非 "central"）
        pos_bottom="5%",  # 适当调整底部间距，避免被容器遮挡
        title_textstyle_opts={  # 明确设置标题文本样式
            "color": "#000000",  # 黑色字体
            "font_size": 24,  # 字体大小
            "font_weight": "bold"  # 加粗显示
        }
    )
)

# 关闭文件
f_us.close()
f_jp.close()
f_in.close()
